

gpus=0

exp=exp/art_v2
log_dir=${exp}/logs
model_save_dir=${exp}/models
sample_dir=${exp}/samples
result_dir=${exp}/results


style_img_dir=/home/wangxin/data/style/wikiart
style_filepath=data/style_train_art.txt
content_img_dir=/home/wangxin/data/detections/coco/train2017/images
content_filepath=data/content_train_coco.txt

CUDA_VISIBLE_DEVICES=${gpus} python3 main.py --clip_model_name=RN50 --crop_size=256 \
--image_size=224 --emb_dim=1024 --batch_size=1 --num_iters=200000 --num_iters_decay=100000 --sample_step=1000 \
--clip_lr=0.001 --g_lr=0.001 --d_lr=0.001 --n_critic=5 --mode=train \
--style_img_dir=${style_img_dir} --style_filepath=${style_filepath} --resume_iters=100000 \
--content_img_dir=${content_img_dir} --content_filepath=${content_filepath}  \
--model_save_dir=${model_save_dir} --sample_dir=${sample_dir} --result_dir=${result_dir} --log_dir=${log_dir}

# --use_patch_clip_loss
# --resume_iters

